from utils.common import pair

try:
    import Image
except ImportError:
    from PIL import Image

# Resize 变换
class Resize:
    def __init__(self, size, mode=Image.BILINEAR):
        """
        将输入的 PIL 图像调整大小到指定的尺寸。

        Args:
            size (int or (int, int)): 所需的输出尺寸。
            mode (int): 插值方法，默认为 Image.BILINEAR。
        """
        self.size = pair(size)
        self.mode = mode

    def __call__(self, img):
        """
        执行 Resize 变换操作。

        Args:
            img (PIL.Image.Image): 待变换的 PIL 图像。

        Returns:
            PIL.Image.Image: 调整大小后的图像。
        """
        return img.resize(self.size, self.mode)

